#pragma once
#include <iostream>
#include <queue>
#include <pthread.h>

using std::queue;

template <class T>
class blockQueue{
    static const int MAXCAP = 10;
public:
    blockQueue(int maxcap = MAXCAP)
        :_maxcap(maxcap)
    {
        pthread_mutex_init(&_mutex, nullptr);
        pthread_cond_init(&_c_cond, nullptr);
        pthread_cond_init(&_p_cond, nullptr);
    }

    T pop(){
        pthread_mutex_lock(&_mutex);
        while(_q.size() == 0){
            pthread_cond_wait(&_c_cond, &_mutex);
        }
        T val = _q.front();
        _q.pop();
        pthread_cond_signal(&_p_cond);  // 消费完可以唤醒生产者
        pthread_mutex_unlock(&_mutex);
        return val;
    }

    void push(const T& val){
        pthread_mutex_lock(&_mutex);
        // 使用whlie循环，防止伪唤醒
        while(_q.size() == _maxcap){
            pthread_cond_wait(&_p_cond, &_mutex);
        }
        _q.push(val);
        pthread_cond_signal(&_c_cond);  // 生产完可以唤醒消费者
        pthread_mutex_unlock(&_mutex);
    }

    ~blockQueue(){
        pthread_mutex_destroy(&_mutex);
        pthread_cond_destroy(&_c_cond);
        pthread_cond_destroy(&_p_cond);
    }
private:
    queue<T> _q;
    pthread_mutex_t _mutex;
    pthread_cond_t _c_cond;
    pthread_cond_t _p_cond;
    int _maxcap;
};